#include <stdio.h> // 标准输入输出库
#include <sys/types.h> // 基本系统数据类型
#include <sys/socket.h> // 套接字库
#include <netinet/in.h> // Internet 地址族
#include <arpa/inet.h> // IP 地址转换库
#include <unistd.h> // UNIX 标准函数库
#include <stdlib.h> // 标准库
#include <string.h> // 字符串操作库
#include <fcntl.h> // 文件控制库
#include <time.h> // 时间库
#include <sys/select.h> // I/O 多路复用
#include <sys/time.h> // 时间库

#define N 64 // 缓冲区大小
#define UDP_PORT 8888 // UDP端口号
#define TCP_PORT 9999 // TCP端口号
#define SERVER_IP "127.0.0.1" // 本地回环地址

int main()
{
    int sockfd, udpfd; // TCP和UDP套接字描述符
    struct sockaddr_in server_addr, udp_addr, client_addr; // 服务器和客户端地址结构体
    char buf[N] = {0}; // 缓冲区
    socklen_t addrlen = sizeof(client_addr); // 客户端地址长度

    // 创建TCP客户端套接字
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0) 
    {
        perror("Socket creation failed"); // 输出错误信息
        return -1; // 返回错误
    }

    // 设置服务器地址
    server_addr.sin_family = AF_INET; // 使用 IPv4
    server_addr.sin_port = htons(TCP_PORT); // 设置端口号
    if (inet_pton(AF_INET, SERVER_IP, &server_addr.sin_addr) <= 0) // 将 IP 地址转换为二进制形式
    {
        perror("Invalid address"); // 输出错误信息
        close(sockfd); // 关闭套接字
        return -1; // 返回错误
    }

    // 连接到服务器
    if (connect(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0)
    {
        perror("Connection to server failed"); // 输出错误信息
        close(sockfd); // 关闭套接字
        return -1; // 返回错误
    }

    printf("Connected to server\n"); // 输出连接成功信息

    // 创建UDP套接字
    udpfd = socket(AF_INET, SOCK_DGRAM, 0);
    if (udpfd < 0)
    {
        perror("UDP socket creation failed"); // 输出错误信息
        close(sockfd);
        return -1;
    }

    // 设置UDP地址
    udp_addr.sin_family = AF_INET;
    udp_addr.sin_port = htons(UDP_PORT);
    udp_addr.sin_addr.s_addr = INADDR_ANY;

    // 绑定UDP套接字
    if (bind(udpfd, (struct sockaddr *)&udp_addr, sizeof(udp_addr)) < 0)
    {
        perror("UDP bind failed"); // 输出错误信息
        close(sockfd);
        close(udpfd);
        return -1;
    }

    int device_online = 0; // 设备在线状态

    while (1)
    {
        // 发送 "Hello, STM32!" 消息
        const char *hello_msg = "Hello, STM32!";
        if (send(sockfd, hello_msg, strlen(hello_msg), 0) < 0) {
            perror("Send failed");
            device_online = 0; // 发送失败，假设设备不在线
        } else {
            printf("Sent: %s\n", hello_msg);
        }

        // 等待响应
        fd_set readfds;
        struct timeval timeout;
        FD_ZERO(&readfds);
        FD_SET(sockfd, &readfds);

        timeout.tv_sec = 5;
        timeout.tv_usec = 0;

        int activity = select(sockfd + 1, &readfds, NULL, NULL, &timeout);

        if (activity > 0 && FD_ISSET(sockfd, &readfds))
        {
            ssize_t len = recv(sockfd, buf, N, 0);
            if (len > 0)
            {
                buf[len] = '\0';
                printf("Received: %s\n", buf);

                if (strcmp(buf, "Yes") == 0)
                {
                    device_online = 1; // 设备在线
                }
            }
            else
            {
                device_online = 0; // 接收失败，假设设备不在线
            }
        }
        else
        {
            device_online = 0; // 超时，假设设备不在线
        }

        // 监听UDP广播

        ssize_t len = recvfrom(udpfd, buf, N, 0, (struct sockaddr *)&client_addr, &addrlen);
        if (len < 0)
        {
            perror("recvfrom failed");
            continue;
        }


        buf[len] = '\0';
        printf("Received UDP message: %s\n", buf);

/*
struct sockaddr_in broadcast_addr;
memset(&broadcast_addr, 0, sizeof(broadcast_addr));
broadcast_addr.sin_family = AF_INET;
broadcast_addr.sin_port = htons(UDP_PORT);
broadcast_addr.sin_addr.s_addr = inet_addr("192.168.0.154"); // 设置为广播地址
*/
        // 检查是否为搜索请求
        if (strcmp(buf, "SEARCH_DEVICE") == 0)
	//if(1)
        {
            const char *response;
            if (device_online) {
                response = "DeviceName|Online|192.168.28.100";
            } else {
                response = "DeviceName|offline|192.168.28.100";
            }
            sendto(udpfd, response, strlen(response), 0, (struct sockaddr *)&client_addr, addrlen);
            printf("Sent response: %s\n", response);
        }

        sleep(5);
    }

    // 关闭套接字
    close(sockfd);
    close(udpfd);

    return 0;
}
